import { Module } from '@nestjs/common';
import { MongooseModule } from '@nestjs/mongoose';
import { ConfigModule } from '@nestjs/config';
import { ComfyUIService } from './comfyui.service';
import { CacheService } from './services/cache.service';
import { GenerationCache, GenerationCacheSchema } from './schemas/generation-cache.schema';

@Module({
    imports: [
        ConfigModule,
        MongooseModule.forFeature([
            { name: GenerationCache.name, schema: GenerationCacheSchema }
        ])
    ],
    providers: [ComfyUIService, CacheService],
    exports: [ComfyUIService]
})
export class ComfyUIModule { } 